package com.langChain4j.api.enums;

import com.base.api.enums.BaseEnum;
import lombok.Getter;

/**
 * @author linjun
 * @date 2025/9/11 10:28
 * @description：模型枚举
 */
@Getter
public enum ModelEnum implements BaseEnum<String> {
    MODEL_QWEN_1_5b("qwen1.5-0.5b-chat", "通义千问1.5b"),
    MODEL_OLLAMA_DEFAULT("llama3.2:1b", "llama3.2:1b"),
    EMBEDDING_MODEL_OLLAMA_MINI_LM("all-minilm", "all-minilm"),

    ;

    private String code;

    private String label;

    ModelEnum(String code, String label) {
        this.code = code;
        this.label = label;
    }

    public String getCode() {
        return code;
    }

    public String getLabel() {
        return label;
    }
}
